#!/usr/bin/python
# -*- coding: utf-8 -*-

'''
3.开发一个注册系统，
页面：
[{name:xxx,age:xxx},xxx]
----------------
*   1-新增用户
*   2-查询用户
*   3-删除用户
----------------
功能1：新增学生信息（姓名和年龄）通过键盘，如果学生已经存在提示用户已存在
功能2：查询学生信息
功能3：删除学生信息
'''
dict_user = {}
dict_student = {}
#菜单首页，打印用户选项
def menu():
    print("*" * 30)
    print("欢迎访问学生管理系统")
    print("*" * 30)
    print("********1:用户登录************")
    print("********2:用户注册************")
    print("********3:用户查询********")
    print("********0:关闭学生管理系统****")

def status_management():
    print("********4:添加学生************")
    print("********5:删除学生************")
    print("********6:查询学生信息********")
    print("********0:关闭学生管理系统****")
    print("*" * 30)

#用户登录方法
def user_install():
    print("请输入用户名和密码进登录.......")
    user_name = input("请输入用户名：")
    password = input("请输入登录密码:")
    for a, b in dict_user.items():
        if a == user_name and b['password'] == password:
            print("登录成功......")
            return 1
        elif a != user_name:
            print("输入用户名不存在，请重新输入......")
            user_install()
        elif a == user_name and b['password'] != password:
            print("输入密码不正确，请重新输入......")
            user_install()

#添加用户方法
def add_user():
    username = input("请输入用户的姓名：")
    password = input("请输入用户的密码：")
    phone = input("请输入用户的联系方式：")
    student_info = {'password': password, 'phone': phone}
    dict_user[username] = student_info
    print(dict_user)
    print("添加成功")

#删除用户方法
def del_user():
    username = input("请输入要删除的用户姓名：")
    if username not in dict_user.keys():
        print("未找到该用户")
        del_user()
    dict_user.pop(username)
    print("您删除了该用户信息")

#查询用户方法
def find_user():
    username = input("请输入要查找的用户姓名:")
    if username not in dict_user.keys():
        print("未找到该用户")
        find_user()
    for username, password in dict_user.items():
        print("用户姓名：{0}, 用户密码：{1}".format(username, password['password']))

#添加学生方法
def add_student():
    id = input("请输入学生的学号：")
    name = input("请输入学生的姓名：")
    age = input("请输入学生的年龄：")
    student_info = {'name': name, 'age': age}
    dict_student[id] = student_info
    print("添加成功")

#删除学生方法
def delete_student():
    id = input("请输入要删除的学生学号：")
    if id not in dict_student.keys():
        print("未找到该学生")
        delete_student()
    dict_student.pop(id)
    print("您删除了该学生信息")

#查询学生方法
def find_student_message():
    print("*" * 30)
    for id, value in dict_student.items():
        print("学号：%s 姓名：%s 年龄：%s 联系方式：%s" % (id, value['name'], value['age'], value['phone']))

#定义主函数，对上面方法进行调用
def main():
    while True:
        menu()
        #用户登录系统
        user_input = input("请输入您的选项：")
        if user_input == '1':
            user_install()
        elif user_input == '2':
            add_user()
        elif user_input == '3':
            find_user()
        elif user_input == '0':
            quit_menu = input("请确认是否要退出系统?(yes/no)")
            if quit_menu == 'yes':
                break
        else:
            print("输入有误请重新输入......")

        #定义一个变量接收登录方法返回：
        install_statu = user_install()
        if install_statu == 1:
            while True:
                status_management()
                user_input = input("请输入您的选项：")
                if user_input == '4':
                    add_student()
                elif user_input == '5':
                    delete_student()
                elif user_input == '6':
                    find_student_message()
                elif user_input == '0':
                    quit_menu = input("请确认是否要退出系统?(yes/no)")
                    if quit_menu == 'yes':
                        break
                else:
                    print("输入有误请重新输入......")

main()


